cmake_minimum_required(VERSION 3.0.0)
project(event2.2 VERSION 0.1.0)

find_package(PythonInterp REQUIRED)

include(CTest)
enable_testing()

add_executable(event2.2 main.cpp)

set(CMAKE_PREFIX_PATH /home/sunqiang/libtorch_gpu/share/cmake/Torch)
set(CMAKE_CUDA_COMPILER /usr/local/cuda/bin/nvcc)
set(CMAKE_CUDA_ARCHITECTURES 86)
find_package(OpenCV REQUIRED)
find_package( Torch REQUIRED )

set(CPACK_PROJECT_NAME ${PROJECT_NAME})
set(CPACK_PROJECT_VERSION ${PROJECT_VERSION})

target_link_libraries(event2.2 ${OpenCV_LIBS} ${TORCH_LIBRARIES})

include(CPack)
